package com.kly.interview.inerceptor;

import com.kly.interview.constant.UserContext;
import com.kly.interview.model.entity.User;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import static com.kly.interview.constant.UserConstant.USER_LOGIN_STATE;

/**
 * @author admin
 * @date 2025/4/28
 */

public class LoginInterceptor implements HandlerInterceptor {



    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
    //     过滤静态方法
        if(request.getMethod().equals("OPTIONS")) {
            return true;
        }
        Object userObj = request.getSession().getAttribute(USER_LOGIN_STATE);
        if(userObj != null) {
            User user = (User) userObj;
            UserContext.setUser(user);
        }


        return true;
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
        HandlerInterceptor.super.postHandle(request, response, handler, modelAndView);
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        UserContext.removeUser();
    }
}
